package com.ww.algorithm.question.leetcode

/**
 * 35.搜索插入位置
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 *
 * 你可以假设数组中无重复元素。
 *
 * 示例 1:
 *
 *     输入: [1,3,5,6], 5
 *     输出: 2
 *
 * 示例 2:
 *
 *     输入: [1,3,5,6], 2
 *     输出: 1
 *
 * 示例 3:
 *
 *     输入: [1,3,5,6], 7
 *     输出: 4
 *
 * 示例 4:
 *
 *     输入: [1,3,5,6], 0
 *     输出: 0
 *
 *
 */
class Solution35 {
    fun search(nums: IntArray, target: Int): Int {
        var left: Int = 0
        var right: Int = nums.size - 1
        // 使用左闭右闭区间
        while (left <= right) {
            var mid = left + (right - left) / 2
            if (nums[mid] > target) {
                right = mid - 1
            } else {
                if (nums[mid] < target) {
                    left = mid + 1
                } else return mid
            }
        }

        return nums.size
    }

    fun search2(nums: IntArray, target: Int): Int {
        var left: Int = 0
        var right: Int = nums.size
        // 使用左闭右开区间
        while (left < right) {
            var mid = left + (right - left) / 2
            if (nums[mid] > target) {
                right = mid
            } else {
                if (nums[mid] < target) {
                    left = mid + 1
                } else return mid
            }
        }

        return right
    }
}

fun main() {
    val solution = Solution35()
    println(solution.search(intArrayOf(1, 2, 3, 4, 5), 5))
    println(solution.search(intArrayOf(1, 2, 3, 4, 5), 2))
    println(solution.search2(intArrayOf(1, 2, 3, 4, 5), 5))
    println(solution.search2(intArrayOf(1, 2, 3, 4, 5), 0))
}